import CONSTANTESBATIMENTS
from Etoile import *
from Planete import *
from Joueur import *
from unite import *
import CONSTANTESUNITES



class Structure(object):
     def __init__(self, type, etoile):
         self.type=type
         self.parent=etoile
         self.vie=CONSTANTESBATIMENTS.structures[type]['vie']
         self.bouclier=CONSTANTESBATIMENTS.structures[type]['bouclier']
         self.technologie=[]
         
     def ajouterTechnologie(self, technologie):
         self.technologie.append(technologie)
     
     def construireUnit(self):
        if (self.type=="stationSpatiale"):
            if(CONSTANTESUNITES.CONSTANTESUNITES.guide['cout']< self.parent.proprietaire.gold):
                self.parent.proprietaire.gold=(self.parent.proprietaire.gold)-(CONSTANTESUNITES.CONSTANTESUNITES.guide['cout'])
                return Unite("guide",self)
        if (self.type=="centreEchange"):
            if(CONSTANTESUNITES.CONSTANTESUNITES.scout['cout']<self.parent.proprietaire.gold):
                self.parent.proprietaire.gold=(self.parent.proprietaire.gold)-(CONSTANTESUNITES.CONSTANTESUNITES.scout['cout'])
                return Unite("scout",self)
                
        if (self.type=="usineAttack"):
            if(CONSTANTESUNITES.CONSTANTESUNITES.rapide['cout']< self.parent.proprietaire.gold):
                self.parent.proprietaire.gold=(self.etoile.proprietaire.gold)-(CONSTANTESUNITES.CONSTANTESUNITES.rapide['cout'])
                return Unite("rapide",self)
        if (self.type=="usineDefense"):
            if(CONSTANTESUNITES.CONSTANTESUNITES.lent['cout']< self.parent.proprietaire.gold):
                self.etoile.proprietaire.gold=(self.etoile.proprietaire.gold)-(CONSTANTESUNITES.CONSTANTESUNITES.lent['cout'])
                return Unite("lent",self)
                
   
class Technologie(object):
    def __init__(self,nom,gold,minerai,energie):
        self.nom=nom
        self.status=False
        self.gold=gold
        self.minerai=minerai
        self.energie=energie
      
class Batiment(object):
     def __init__(self, type, planete):
         self.type=type
         self.parent=planete
         self.vie=CONSTANTESBATIMENTS.batiments[type]['vie']
         self.bouclier=CONSTANTESBATIMENTS.batiments[type]['bouclier']
         self.vitesse=CONSTANTESBATIMENTS.batiments[type]['vitesse']
         if (self.type == 'tour'):
             self.attaque = CONSTANTESBATIMENTS.batiments[type]['attaque'] 
         self.technologie=[]
         
     def ajouterTechnologie(self, technologie):
         self.technologie.append(technologie)
        
